comprehension(生成式)

• 可以使用在可迭代的物件上，最常用在 list、dict、set 上
• 搭配前面的 lambda(匿名函式) 可以使用一行程式來完成比較複雜的程式邏輯，有興趣的可以進一步參考這本書 Python不廢話，一行程式碼：像高手般寫出簡潔有力的Python程式碼

list comprehension

• 語法：[ 運算式 for 變數(item) in 可迭代的物件(iterable) if 條件]
  • if 條件非必須
  • 可迭代的物件為原本的 list
  • 運算式也可加入 if 判斷

x = [1, 2, 3, 4]
x_list = [item * 2 for item in x]
print(x_list)
# [2, 4, 6, 8]

x_list = [item * 2 for item in x if item % 2 == 0]
print(x_list)
# [4, 8]

x_list = [item if item > 2 else 0 for item in x ]
print(x_list)
# [0, 0, 3, 4]

dict comprehension

• 語法：{ key 運算式:value 運算式 for 變數(item) in 可迭代的物件(iterable) if 條件}
  • if 條件非必須
  • 可迭代的物件為原本的 list
  • 運算式也可加入 if 判斷

x_dict = {item:item * 2 for item in x}
print(x_dict)
# {1: 2, 2: 4, 3: 6, 4: 8}

x_dict = {item:item * 2 for item in x if item > 2}
print(x_dict)
# {3: 6, 4: 8}

x_dict = {f"{item}":item if item > 2 else 0 for item in x }
print(x_dict)
# {'1': 0, '2': 0, '3': 3, '4': 4}

generator(產生器)

• 當把 list 生成式的 [] 變更為 ()，並不是 truple，而是一個 generator(產生器)
• 可用迴圈逐一印出

x = [1, 2, 3, 4]
x_list = [item * 2 for item in x]
print(x_list)
# [2, 4, 6, 8]

x_gen = (item * 2 for item in x)
print(x_gen)
# generator object

參考資料

• [Python教學]Python Comprehension語法應用教學
• 生成式 ( 串列、字典、集合、元組 )
• Python 推导式
• Python Comprehension 解析式語法

次回

來看看 type hint 吧！